# -*- coding: utf-8 -*-

from keybert import KeyBERT

class KeyBERTExtractor:
    def __init__(self):
        self.kb = KeyBERT()

    def extract_keywords(self, text, top_n=5):
        """提取文本的关键词"""
        keywords = self.kb.extract_keywords(text, top_n=top_n)
        return keywords

# 使用示例
if __name__ == "__main__":
    text = "我爱自然语言处理和机器学习。机器学习是一种数据分析方法，它使用算法来从数据中学习。"
    
    # 创建 KeyBERT 提取器实例
    extractor = KeyBERTExtractor()
    
    # 提取关键词
    keywords = extractor.extract_keywords(text)
    print("提取的关键词：", keywords)
